#include <stdio.h>
#include "bootpack.h"

#define PORT_KEYDAT 0x0060
#define PORT_KEYSTA 0x0064
#define PORT_KEYCMD 0x0064
#define KEYSTA_SEND_NOTREADY 0x02
#define KEYCMD_WRITE_MODE 0x60
#define KBC_MODE 0x47			 /* KBC - 键盘控制电路keyboard controller */
#define KEYCMD_SENDTO_MOUSE 0xd4 /* 如果往键盘控制电路发送指令0xd4，下一个数据就会自动发送给鼠标 */
#define MOUSECMD_ENABLE 0xf4

extern struct FIFO8 keyfifo;
extern struct FIFO8 mousefifo;

void wait_KBC_sendready(void);
void init_keyboard(void);
void enable_mouse(void);

void HariMain(void)
{
	struct BOOTINFO *binfo = (struct BOOTINFO *)0x0ff0;
	char s[40], mcursor[256], keybuf[32], mousebuf[128];
	int mx, my, i, j;

	init_gdtidt();
	init_pic();
	io_sti(); /* 由于IDT/PIC初始化已完成，因此CPU中断被禁用。 */
	fifo8_init(&keyfifo, 32, keybuf);
	fifo8_init(&mousefifo, 128, mousebuf);
	/* 修改IMR（中断屏蔽寄存器），如果某一位的值是1，则该位所对应的IRO信号被屏蔽,PIC就忽视该路信号 */
	io_out8(PIC0_IMR, 0xf9); /* 允许PIC1和键盘（11111001） */
	io_out8(PIC1_IMR, 0xef); /* 允许鼠标（11101111） */

	init_keyboard();

	init_palette();
	init_screen8(binfo->vram, binfo->scrnx, binfo->scrny);
	mx = (binfo->scrnx - 16) / 2; /* 屏幕中心坐标计算 */
	my = (binfo->scrny - 28 - 16) / 2;
	init_mouse_cursor8(mcursor, COL8_008484);
	putblock8_8(binfo->vram, binfo->scrnx, 16, 16, mx, my, mcursor, 16);
	sprintf(s, "(%d, %d)", mx, my);
	putfonts8_asc(binfo->vram, binfo->scrnx, 0, 0, COL8_FFFFFF, s);

	unsigned char mouse_dbuf[3], mouse_phase; /* mouse_dbuf记录数据，mouse_phase记录接收鼠标数据的工作进展到的阶段 */
	enable_mouse();
	mouse_phase = 0; /* 进入到等待鼠标的0xfa的状态 */

	for (;;)
	{
		io_cli();
		if (fifo8_status(&keyfifo) + fifo8_status(&mousefifo) == 0)
		{
			io_stihlt();
		}
		else
		{
			if (fifo8_status(&keyfifo) != 0)
			{
				i = fifo8_get(&keyfifo);
				io_sti();
				sprintf(s, "%02X", i);
				boxfill8(binfo->vram, binfo->scrnx, COL8_008484, 0, 16, 15, 31);
				putfonts8_asc(binfo->vram, binfo->scrnx, 0, 16, COL8_FFFFFF, s);
			}
			else if (fifo8_status(&mousefifo) != 0)
			{
				i = fifo8_get(&mousefifo);
				io_sti();
				/* 先把最初读到的0xfa舍弃掉，之后每次从鼠标送来的数据是3个字节一组的，所以每积累到3个字节就显示在屏幕上 */
				if (mouse_phase == 0)
				{
					/* 等待鼠标的0xfa的状态 */
					if (i == 0xfa)
					{
						mouse_phase = 1;
					}
				}
				else if (mouse_phase == 1)
				{
					/* 等待鼠标的第一字节 */
					mouse_dbuf[0] = i;
					mouse_phase = 2;
				}
				else if (mouse_phase == 2)
				{
					/* 等待鼠标的第二字节 */
					mouse_dbuf[1] = i;
					mouse_phase = 3;
				}
				else if (mouse_phase == 3)
				{
					/* 等待鼠标的第三字节 */
					mouse_dbuf[2] = i;
					mouse_phase = 1;
					/* 鼠标的3个字节都齐了，显示出来 */
					sprintf(s, "%02X %02X %02X", mouse_dbuf[0], mouse_dbuf[1], mouse_dbuf[2]);
					boxfill8(binfo->vram, binfo->scrnx, COL8_008484, 32, 16, 32 + 8 * 8 - 1, 31);
					putfonts8_asc(binfo->vram, binfo->scrnx, 32, 16, COL8_FFFFFF, s);
				}
			}
		}
	}
}

void wait_KBC_sendready(void)
{
	/* 等待键盘控制电路准备完毕 */
	for (;;)
	{
		if ((io_in8(PORT_KEYSTA) & KEYSTA_SEND_NOTREADY) == 0)
		{
			break;
		}
	}
	return;
}

void init_keyboard(void)
{
	/* 初始化键盘控制电路 */
	wait_KBC_sendready();
	io_out8(PORT_KEYCMD, KEYCMD_WRITE_MODE);
	wait_KBC_sendready();
	io_out8(PORT_KEYDAT, KBC_MODE); /* 模式设定的指令是0x60，利用鼠标模式的模式号码是0x47 */
	return;
}

void enable_mouse(void)
{
	/* 激活鼠标 */
	wait_KBC_sendready();
	io_out8(PORT_KEYCMD, KEYCMD_SENDTO_MOUSE);
	wait_KBC_sendready();
	io_out8(PORT_KEYDAT, MOUSECMD_ENABLE);
	return; /* 顺利的话，键盘控制器会返送回ACK(0xfa) */
}